import numpy as py
# Config.py 定义了项目所需要用到的参数
class config():
    # 指定手写体的路径，数据存放的位置的路径
    data_path = './mnist_data'
    # 定义训练神经网络时遍历整个训练集的次数
    epochs = 50     #训练次数
    # 用于指定在可视化时要显示的样本数量
    plot_num = 10
    classes = 10
    batch_size = 32
    learning_rate = 0.0001  #学习率
    model_name = 'mnist_cnn'    #模型名称
    save_path = './{}.pth'.format(model_name)